package cn.itcast.mq.config;

import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.*;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.support.converter.Jackson2JsonMessageConverter;
import org.springframework.amqp.support.converter.MessageConverter;
import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
@Slf4j
public class CommonConfig implements ApplicationContextAware {

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        // 从ioc容器中 拿到template对象
        RabbitTemplate rabbitTemplate = applicationContext.getBean(RabbitTemplate.class);

        // 设置发送者确认回调函数
        rabbitTemplate.setConfirmCallback(new RabbitTemplate.ConfirmCallback() {
            /**
             * @param correlationData  自定义的数据   消息UUID
             * @param ack   是否确认:   true 消息发送到exchange中    false 消息未发送到exchange
             * @param cause  原因:
             */
            @Override
            public void confirm(CorrelationData correlationData, boolean ack, String cause) {
                log.info("发送确认回调触发  消息的ID ==> {}",correlationData.getId());
                if(ack){
                    log.info("消息成功发送到交换机中 ~！！！！");
                }else {
                    log.error("消息发送到交换机失败 ~！！！！   原因: {}",cause);
                    // 消费未投递到交换机 ， 可以触发   重发逻辑
                }
            }
        });
        rabbitTemplate.setReturnCallback(new RabbitTemplate.ReturnCallback() {
            /**
             * 只要这个方法触发，代表消息没能正确的路由到队列，被mq 返还回来了
             * @param message   返回的消息
             * @param replyCode   回复状态码:
             * @param replyText   回复内容
             * @param exchange    交换机
             * @param routingKey  路由key
             */
            @Override
            public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
                // 投递失败，记录日志
                log.info("消息发送失败，应答码{}，原因{}，交换机{}，路由键{},消息{}",
                        replyCode, replyText, exchange, routingKey, message.toString());
                // 如果有业务需要，可以重发消息
            }
        });
    }


//    @Bean
//    public DirectExchange simpleExchange(){
//        // 三个参数：交换机名称、是否持久化、当没有queue与其绑定时是否自动删除
//        return new DirectExchange("simple.direct");
//    }
//    @Bean
//    public Queue simpleQueue(){
//        return new Queue("simple.queue");
//    }
//    @Bean
//    public Binding binding(){
//        return BindingBuilder.bind(simpleQueue()).to(simpleExchange()).with("simple");
//    }

    @Bean
    public Queue lazyQueue(){
        return QueueBuilder.durable("lazy.queue")
                .lazy()
                .build();
    }
    @Bean
    public Queue normalQueue(){
        return QueueBuilder.durable("normal.queue")
                .build();
    }
}
